package com.sxj.corejava.code12_jdbc;

import com.sxj.corejava.util.DataAccessException;
import com.sxj.corejava.util.JDBCUtil;

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;

/**
 * @author 石小俊
 * @date 2024年07月02日 10:30
 */
public class Test06_保存返回主键 {

    public static void main(String[] args) throws DataAccessException {
        User user = new User("lucy", "666666", "13912345677", "江苏-无锡");
        int id = insertReturnPrimaryKey(user);
        System.out.println("保存后返回的主键的值为:" + id);
    }

    public static Integer insertReturnPrimaryKey(User user) throws DataAccessException {
        Connection conn = null;
        PreparedStatement ps = null;
        ResultSet rs = null;
        Integer id = null;

        try {
            conn = JDBCUtil.getConnection();
            String sql = new StringBuffer()
                    .append(" insert into ")
                    .append(" t_user ")
                    .append("   (username,password,phone,address)")
                    .append(" values ")
                    .append("   (?,?,?,?)")
                    .toString();
            // 在获取状态集时,可以通过参数二表示是否返回自动生成的键
            // 其值有两种:
            // 1.Statement.RETURN_GENERATED_KEYS:返回
            // 2.Statement.NO_GENERATED_KEYS:不返回
            // PreparedStatement是Statement的子类,可以调用父类常量
            ps = conn.prepareStatement(sql, PreparedStatement.RETURN_GENERATED_KEYS);
            ps.setString(1, user.getUsername());
            ps.setString(2, user.getPassword());
            ps.setString(3, user.getPhone());
            ps.setString(4, user.getAddress());
            // 执行DML操作
            ps.executeUpdate();
            // 在数据库中,只有执行了DML操作之后,才会去生成对应的主键
            // 执行之后开始获取生成的主键的值
            rs = ps.getGeneratedKeys();
            // 此时的结果集中存在所有自动生成的键的值
            // 而所有自动生成的键的列名是由数据库自己定义的
            // 字段名与我们表中的字段名是不一样的
            // 但是,我们可以确定,此时生成的只有主键一列
            // 因此,我们可以通过第几列来获取
            if (rs.next()) {
                id = rs.getInt(1);
            }
        } catch (DataAccessException e) {
            e.printStackTrace();
            throw e;
        } catch (SQLException e) {
            e.printStackTrace();
            throw new DataAccessException("数据访问异常");
        } finally {
            JDBCUtil.close(conn, ps, rs);
        }

        return id;
    }

}
